package leetcode;

/*
347. 前 K 个高频元素
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
提示：
你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。
*/

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.PriorityQueue;
import java.util.TreeMap;

public class problems_347 {
    public static void main(String[] args) {
        System.out.println(new Solution().topKFrequent(new int[]{1, 1, 1, 2, 2, 3}, 2));
    }
    static class Solution {
        public int[] topKFrequent(int[] nums, int k) {
            // 转为map，统计去重
            TreeMap<Integer, Integer> map = new TreeMap<>();
            for (int i = 0; i < nums.length; i++) {
                map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
            }
            // value排序，取得目标值
            Object[] arrValue = map.values().stream().sorted().toArray();
            if(k > arrValue.length) return new int[]{};
            Integer targetValue = (Integer)arrValue[arrValue.length - k];

            List<Integer> list = new ArrayList<>();
            for (Integer key: map.keySet()){
                if(map.get(key) >= targetValue) list.add(key);
            }
            return list.stream().mapToInt(Integer::valueOf).toArray();
        }
    }
}